// #include<iostream>
#include<string.h>
#include<sys/wait.h>
#include<sys/stat.h>
#include<fcntl.h>
#include<ctype.h>
#include<sys/types.h>
#include<unistd.h>
#include<stdlib.h>
#include<stdio.h>

#define SIZE 1024
#define MAX_ARGS 64
#define SEP " "
char* argv[MAX_ARGS];
char pwd[SIZE];
char env[SIZE];
#define STREND '\0'
int lastcode = 0;


//下面都和重定向相关
#define NoneRedir -1
#define StdinRedir 0
#define StdoutRedir 1
#define AppendRedir 2
#define IgnSpace(buf,pos) do{ while(isspace(buf[pos])) pos++; }while(0) //跳过空格


int redir_type = NoneRedir;
char* filename = NULL;
const char* Username()
{
    char * username = getenv("USER");
    if(username) return username;
    else return "None";
}
const char* HostName()
{
    char * hostname = getenv("HOSTNAME");
    if(hostname) return hostname;
    else return "None";
}
const char* CurrentDir()
{
    char * dir = getenv("PWD");
    if(dir) return dir;
    else return "None";
}
char* Home()
{
    return getenv("Home");
}
int Interactive(char out[],int size)
{
    printf("[%s@%s %s]$ ",Username(),HostName(),CurrentDir());
    fgets(out,size,stdin);
    out[strlen(out)-1] = 0;//'\0' , 输入为空串
    return strlen(out);
}

void CheckRedir(char in[])
{
    // ls -a -l
    // ls -a -l > log.txt 输出重定向
    // ls -a -l >> log.txt 追加重定向
    // cat < log.txt 输入重定向
    redir_type = NoneRedir;
    filename = NULL;
    int pos = strlen(in)-1;
    while(pos)
    {
        if(in[pos] == '>')
        {
            if(in[pos-1] == '>')
            {
                //追加重定向
                redir_type = AppendRedir;
                //目的找出重定向文件名
                in[pos-1] = STREND;
                pos++;
                IgnSpace(in,pos);//跳过空格
                filename = in+pos;
                break;
            }
            else 
            {
                //输出重定向
                redir_type = StdoutRedir;
                in[pos] = STREND;
                pos++;
                IgnSpace(in,pos);
                filename = in+pos;
                break;
            }
        }
        else if(in[pos] == '<')
        {
            //输入重定向
            redir_type = StdinRedir;
            in[pos++] = STREND;
            IgnSpace(in,pos);
            filename = in + pos;
            break;
        }
        else {
            pos--;
        }
    }
}
void Split(char in[])
{
    CheckRedir(in);
    int i = 0;
    argv[i++] = strtok(in,SEP);
    while(argv[i++] = strtok(NULL,SEP));
    if(strcmp(argv[0],"ls") == 0)
    {
        argv[i-1] = (char*)"--color";
        argv[i] = NULL;
    }
}
int BuildinCmd()
{
    int ret = 0;
    //处理内建命令，是 ret = 1，否 ret = 0;
    if(strcmp("cd",argv[0]) == 0)
    {
        //2. 执行 cd xxxx or cd 
        ret = 1;
        printf("%s\n",argv[1]);
        char* target = argv[1];
        if(!target) target = Home();
        chdir(target);//改变工作目录
        //将改变后的工作目录添加到环境变量中
        char buffer[1020];
        getcwd(buffer,1020);//获取当前路工作路径
        snprintf(pwd, SIZE, "PWD=%s",buffer);//格式化
        putenv(pwd);//写入环境变量
    }
    else if(strcmp("export",argv[0]) == 0)
    {
        ret = 1;
        if(argv[1])
        {
            strcpy(env,argv[1]);
            putenv(env);
        }
    }
    else if(strcmp("echo",argv[0]) == 0)
    {
        ret = 1;
        if(argv[1] == NULL)
        {
            printf("\n");
        }
        else
        {
            if(argv[1][0] == '$')
            {
                if(argv[1][1] == '?')
                {
                    //此时查看进程推出码
                    printf("%d\n",lastcode);
                    lastcode = 0;
                }
                else 
                {
                    //查看环境变量
                    char* e = getenv(argv[1]+1);
                    if(e) printf("%s\n",e);
                }
            }
            else 
            {
                printf("%s\n",argv[1]);
            }
        }
    }
    return ret;
}
void Excute()
{
    mode_t mode = 0644;
    pid_t id = fork();
    if(id == 0)
    {
        //程序替换，执行子进程
        int fd = -1;
        if(redir_type == StdinRedir)
        {
            //只读打开文件，将这个文件设置为程序的标准输入
            fd = open(filename, O_RDONLY);
            dup2(fd,0);
        }
        else if(redir_type == StdoutRedir)
        {
            //打开写方式文件，将这个文件作为程序的标准输出，清空写入
            fd = open(filename,O_CREAT|O_WRONLY|O_TRUNC,mode);
            dup2(fd,1);
        }
        else if(redir_type == AppendRedir)
        {
            //打开文件，将这个文件作为输出，追加写入
            fd = open(filename,O_CREAT|O_WRONLY|O_APPEND,mode);
            dup2(fd,1);
        }
        else 
        {

        }
        execvp(argv[0],argv);
        exit(1);
    }
    int status = 0;
    pid_t rid = waitpid(id,&status,0);//回收子进程资源，防止僵尸进程
    if(rid == id) lastcode = WEXITSTATUS(status);
}
int main()
{
    while(1)
    {
        char commandline[SIZE];
        //1. 打印命令提示符，获取用户输入命令行字符串
        int n = Interactive(commandline,SIZE);
        if(n == 0) continue;
        //2. 对字符穿进行切割
        Split(commandline);
        //3. 处理内建指令
        n = BuildinCmd();
        if(n) continue;
        Excute();
    }
    return 0;
}
